- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 52 for _output_shapes (0.18 sec)
-
tensorflow/compiler/mlir/lite/tests/pin-ops-with-side-effects.mlir
func.func @tf_if_gets_control_node(%arg0: tensor<1xi1>)->() { "tf.If"(%arg0) {_lower_using_switch_merge = true, _read_only_resource_inputs = [], device = "", else_branch = @noop, is_stateless = false, output_shapes = [#tf_type.shape<>], then_branch = @noop} : (tensor<1xi1>) -> () func.return } // CHECK-NEXT: %[[CONTROL:.*]] = tfl.control_node controls "tf.If" // CHECK-NEXT: return // CHECK-LABEL: @tfl_if_gets_control_node
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 10:45:19 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/func_list_attr.mlir
// CHECK-NEXT: } // CHECK-NEXT: } // CHECK-NEXT: } // CHECK: } %1:2 = tf_executor.island wraps "tf.Case"(%0#0) {Tin = [], Tout = ["tfdtype$DT_FLOAT"], branches = [@foo, @bar], device = "", output_shapes = [], is_stateless = false} : (tensor<i32>) -> tensor<*xf32> loc("Case") tf_executor.fetch } func.return } // CHECK-DAG: name: "foo" func.func @foo() -> tensor<10xf32> { %0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape, const SmallVector<int64_t, 4>& output_shape) { auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); SmallVector<mlir::Type, 2> input_types{input_type, indices_type};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/analyze-variables.mlir
output_shapes = [#tf_type.shape<?>], output_types = [!tf_type.string]} : (tensor<*x!tf_type.variant>, tensor<i64>) -> tensor<!tf_type.variant> %1 = "tf.ReduceDataset"(%0, %cst_1, %arg0) { Targuments = [!tf_type.resource], Tstate = [i32], device = "", f = @__reduce_func, f._tf_data_function = true, output_shapes = [#tf_type.shape<>],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 09 11:49:28 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h
// Converts a shape from MLIR to a TensorFlow tensor shape proto. void ConvertToTensorShapeProto(llvm::ArrayRef<int64_t> shape, TensorShapeProto* output_shape); // Converts an MLIR type to a TensorFlow tensor shape. PartialTensorShape ConvertTypeToTensorShape(const mlir::Type& type); // Converts an MLIR shaped type to a TensorFlow shape attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/functional-while-ops.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 15 19:42:47 UTC 2021 - 5.5K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.cc
saved_concrete_function.output_signature().tensor_spec_value(); PartialTensorShape output_shape = output_tensor_spec_proto.shape(); DataType output_dtype = output_tensor_spec_proto.dtype(); TaggedValue output_tensor_spec(output_shape, output_dtype); // Register the function trace. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 7.4K 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/tensorflow/tests/graphdef2mlir/graph-input-func-arg-name-collision.pbtxt
name: "while_body_6" } } } attr { key: "cond" value { func { name: "while_cond_5" } } } attr { key: "output_shapes" value { list { shape { } shape { } shape { } } } } attr { key: "parallel_iterations"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 11 19:14:04 UTC 2020 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
// // ```mlir // %1 = "tf.DeviceIndex"() // {device = "", device_names = ["CPU", "GPU"]} : () -> tensor<i32> // %4 = "tf.Case"(%1, %arg0, %arg1) // {branches = [@foo, @baz], output_shapes = [#tf_type.shape<>]} : // (tensor<i32>, tensor<f32>, tensor<f32>) -> tensor<f32> // ``` // // Shows an example where there are 2 different functions which could be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0)