- Sort Score
- Result 10 results
- Languages All
Results 81 - 88 of 88 for input_shapes_ (0.68 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Set dimension sizes specified by broadcast_dimensions. ArrayRef<int64_t> input_shape = mlir::cast<ShapedType>(input.getType()).getShape(); for (auto x : llvm::enumerate(broadcast_dimensions)) { expanded_shape[x.value().getSExtValue()] = rewriter.getI64IntegerAttr(input_shape[x.index()]); } // Create the expanded type wrapped in a arith::ConstantOp. auto attr_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
// CHECK-DAG: [[PADDINGS_SUM:%.+]] = "tf.AddV2"([[PADDINGS]]#0, [[PADDINGS]]#1) // CHECK-DAG: [[INPUT_SHAPE:%.+]] = "tf.Const"() <{value = dense<[3, 5, 7, 10]> : tensor<4xi64>}> // CHECK-DAG: [[PADDED_SHAPE:%.+]] = "tf.AddV2"([[PADDINGS_SUM]], [[INPUT_SHAPE]]) // CHECK-DAG: [[PADDED_SHAPE_SPLITS:%.+]]:4 = "tf.Split"([[ZERO_I32]], [[PADDED_SHAPE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
specs_.unconditionally_use_set_output_shapes; for (const auto& name_and_value : func_def->attr()) { if (name_and_value.first == "_input_shapes") { auto& list = name_and_value.second.list(); auto& signature = func_def->signature(); // Some models have "_input_shapes" attribute, but with its value empty if (list.shape_size() > 0 && list.shape_size() != signature.input_arg_size()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
input_shape[1], block_shape[0], ..., input_shape[M], block_shape[M-1], input_shape[M+1], ..., input_shape[N-1]] 3. Reshape `permuted` to produce `reshaped_permuted` of shape [batch / prod(block_shape), input_shape[1] * block_shape[0], ..., input_shape[M] * block_shape[M-1], input_shape[M+1], ..., input_shape[N-1]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/cc/saved_model/testdata/chunked_saved_model/chunked_model/saved_model.pbtxt
} attr { key: "_construction_context" value { s: "kEagerRuntime" } } attr { key: "_input_shapes" value { list { shape { dim { size: 150 } dim { size: 150 }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 21:43:11 UTC 2023 - 531.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
"tf.PwStreamResults"(%arg0, %arg1) {_callback_id = -2694175233261920887 : i64, _controller_address = "[2002:afb:afb::]:10004",...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
%paddings = "tf.Const"() {value = dense<[[1, 2], [1, 2]]> : tensor<2x2xi64>} : () -> tensor<2x2xi64> // expected-error @+1 {{requires block_shape[i] divides input_shape[i + 1] + paddings[i, 0] + paddings[i, 1]; failed for i=1}} %1 = "tf.SpaceToBatchND"(%input, %block_shape, %paddings) : (tensor<3x5x7x10xf32>, tensor<2xi64>, tensor<2x2xi64>) -> tensor<36x2x3x10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)