- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 95 for _output_shapes (0.26 sec)
-
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) -
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) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
def FuseBinaryOpWithTransposeConv#binaryOp : Pat< (binaryOp (TFL_TransposeConvOp:$output $output_shape, $weights, $input, (Arith_ConstantOp FloatElementsAttr:$bias), $padding, $stride_h, $stride_w, TFL_AF_None), (Arith_ConstantOp FloatElementsAttr:$value), $act_fn), (TFL_TransposeConvOp $output_shape, $weights, $input, (binaryOp (Arith_ConstantOp $bias),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
auto new_shape = rewriter.create<TF::ShapeOp>(loc, shape_type, input); SmallVector<int64_t, 8> output_shape(/*Size=*/1, op.getNumElements()); for (const auto &dim : dense_elem_attr.getValues<APInt>()) output_shape.push_back(dim.getSExtValue()); RankedTensorType result_type = tensorflow::GetTypeFromTFTensorShape( output_shape, getElementTypeOrSelf(input)); rewriter.replaceOpWithNewOp<TF::ReshapeOp>(op, result_type, input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)