- Sort Score
- Result 10 results
- Languages All
Results 21 - 28 of 28 for PartitionedCall (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/remove_unused_arguments.mlir
return %arg0 : tensor<f32> } // CHECK-LABEL: handles_partitioned_function_calls func.func @handles_partitioned_function_calls(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> { // CHECK: PartitionedCall"() %1 = "tf.PartitionedCall"(%arg0, %arg1) {f = @f} : (tensor<f32>, tensor<f32>) -> tensor<f32> return %1 : tensor<f32> } // ----- func.func private @f(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 06 23:00:44 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
Output in = ops::Placeholder(root, DT_FLOAT); Output perm = ops::Const(root, {3, 1, 2, 0}); NameAttrList b_name_attr; b_name_attr.set_name("TransposeCall"); ops::PartitionedCall call(root.WithOpName("call"), {in, perm}, {DT_FLOAT}, b_name_attr); call.output.front().node()->AddAttr(kXlaMustCompileAttr, true); std::unique_ptr<Graph> graph;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.cc
auto value = device_attr.getValue(); // TODO(b/229028654): Remove string conversion once we have C++17. absl::string_view device_type(value.data(), value.size()); // Device type may be empty for some ops, e.g. tf.PartitionedCall. auto it = std::find(kValidDeviceTypes.begin(), kValidDeviceTypes.end(), device_type); if (it == kValidDeviceTypes.end()) return failure(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
return failure(); } rewriter.setInsertionPointAfter(call_op); // The TPUPartitionedCall has a TPUOrdinalSelectorOp for its last argument // which should be removed. So the replaced PartitionedCall op should keep // its original arguments except for the last element. SmallVector<Value> args = call_op.getOperands().drop_back(); rewriter.replaceOpWithNewOp<TF::PartitionedCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/ops/xla_ops.cc
.SetShapeFn([](InferenceContext* c) { c->set_output(0, c->input(0)); return absl::OkStatus(); }) .Doc(R"(XLA Merge Op. For use by the XLA JIT only. Merges the outputs from the PartitionedCall node and the _XlaRun node. Unlike the TensorFlow Merge op, which requires inputs of some types to be placed on the host, the _XlaMerge op can merge inputs of all types when
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 09:08:06 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
// CHECK-NEXT: [[q_bias:%.+]] = "quantfork.qcast"([[bias]]) : (tensor<2xf32>) -> tensor<2x!quant.uniform<i32:f32, 0.044022349891595126>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
return true; } } return false; } // Checks if the function is only used by supported ops. Returns false when the // function has no uses. Currently, only PartitionedCall is supported. // TODO(b/284222309): Support lifting for functions called by control flow. bool UsedBySupportedOps(ModuleOp module, func::FuncOp func) { auto function_uses =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0)