- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 120 for _output_shapes (0.35 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
int64_t out_row_dim = output_shape[output_shape.size() - 2]; int64_t out_col_dim = output_shape[output_shape.size() - 1]; int64_t expected_out_row_dim = op.getAdjX() ? x_col_dim : x_row_dim; int64_t expected_out_col_dim = op.getAdjY() ? y_row_dim : y_col_dim; if (expected_out_row_dim != ShapedType::kDynamic && out_row_dim != ShapedType::kDynamic && out_row_dim != expected_out_row_dim)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K 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/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
} // Gather shapes for output. for (auto v : ddn.lhs_batch_dimensions()) { output_shape.push_back(lhs_shape[v]); } // Batch dimension is gathered from the right side. if (output_shape.empty()) { for (auto v : ddn.rhs_batch_dimensions()) { output_shape.push_back(rhs_shape[v]); } } // Gather remaining dimensions.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K 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/functional-control-flow-to-regions.mlir
device = "/job:tpu_host_worker/replica:0/task:0/device:CPU:0", finalize_func = @finalize, init_func = @init, next_func = @next, operandSegmentSizes = array<i32: 1, 2, 1>, output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string], metadata = ""} : ( tensor<4xf32>, tensor<3xf32>, tensor<!tf_type.resource>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 21:59:28 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
%2 = "tf.Const"() {__op_key = 2: i32, device = "/device:CPU:0", value = dense<1> : tensor<i64>} : () -> tensor<i64> %3 = "tf.RangeDataset"(%0, %1, %2) {__op_key = 3: i32, device = "/device:CPU:0", output_shapes = [#tf_type.shape<>], output_types = [i64], metadata = ""} : (tensor<i64>, tensor<i64>, tensor<i64>) -> tensor<!tf_type.variant> // CHECK: tf_mlrt.executeop{{.*}}op: \22FlatMapDataset\22
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
continue; } shape.dims = new int64_t[shape.num_dims]; for (size_t j = 0; j < shape.num_dims; ++j) { shape.dims[j] = c.Value(c.Dim(shape_handle, j)); } } if (output_shapes != nullptr) *output_shapes = output_shapes_result; // TODO(bgogul): Set output_resource_shapes_and_types. } void TF_ImportGraphDefOptionsSetValidateColocationConstraints(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} auto output_shape = xla::ShapeInference::InferGatherShape( input_shape, start_indices_shape, gather_dim_numbers, slice_sizes); if (!output_shape.ok()) { op->emitError() << output_shape.status().message(); return false; } auto refined_type = xla::ConvertShapeToType<RankedTensorType>( *output_shape, mlir::Builder(op)); if (!refined_type.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)