- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for input_shapes_ (0.16 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
Output input_shape_i = ops::Slice( host_scope.WithOpName("input_shape_", i), input_shape, constant_pool.Get1DHostConstant(i), constant_pool.Get1DHostConstant(1)); slice_size.push_back(ops::Sub(host_scope.WithOpName("slice_size_", i), input_shape_i, begin_i)); DCHECK_EQ(slice_size.back().type(), DT_INT64); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
// Set input_shapes. for (int i = 0; i < num_inputs; ++i) { std::vector<DimensionHandle> dims; const TF_ShapeAndType& input_shape = input_shapes->items[i]; if (input_shape.num_dims == InferenceContext::kUnknownRank) { c.SetInput(i, c.UnknownShape()); continue; } dims.reserve(input_shape.num_dims); for (int j = 0; j < input_shape.num_dims; ++j) {
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/translate/tf_mlir_translate.cc
const std::vector<std::optional<std::vector<int>>>& input_shapes, const std::vector<std::string>& output_arrays, const std::vector<std::string>& control_output_arrays, const GraphdefToMlirOptions& import_options, mlir::MLIRContext* context) { auto module_or = GraphdefToMlirImport( input, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, import_options, context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 11:51:44 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { auto input = conv2d.getInput(); auto input_shape = mlir::cast<RankedTensorType>(input.getType()).getShape(); SmallVector<int64_t, 4> transform_shape = { input_shape[0], input_shape[1] / block_size, input_shape[2] / block_size, input_shape[3] * block_size * block_size}; auto transform_result_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
Value &padding, int num_dims) { ShapedType input_shape = mlir::cast<ShapedType>(input.getType()); SmallVector<int64_t> spatial_dims(num_dims - 2); absl::c_iota(spatial_dims, 1); bool has_dynamic_spatial_dim = absl::c_any_of( spatial_dims, [&input_shape](int64_t dim) { return input_shape.isDynamicDim(dim); }); if (conv_padding.strref() == "SAME" && has_dynamic_spatial_dim) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
StringAttr conv_padding, ArrayAttr explicit_paddings) { auto input_shape = mlir::cast<ShapedType>(input.getType()); auto filter_shape = mlir::cast<ShapedType>(filter.getType()); if (!input_shape.hasRank() || input_shape.getRank() != 4 || !filter_shape.hasRank() || filter_shape.getRank() != 4) { emitError(loc, "input and filter are expected to be 4D tensors");
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/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
arg_kinds_vector)) { XlaArgument& arg = std::get<0>(arg_components); TensorShape shape; auto input_shapes = std::get<1>(arg_components); if (input_shapes.has_value()) { TF_RETURN_IF_ERROR(TensorShapeUtils::MakeShape(*input_shapes, &shape)); } else { TF_RETURN_IF_ERROR( TensorShapeUtils::MakeShape(static_cast<int*>(nullptr), 0, &shape)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
if (input_shape.size() != perm_values.size()) { return false; } SmallVector<int, 8> old_major_index_ordering; SmallVector<int, 8> new_major_index_ordering; for (int i = 0, end = input_shape.size(); i < end; i++) { if (input_shape[i] != 1) { old_major_index_ordering.push_back(i); } if (input_shape[perm_values[i]] != 1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
signatures=model.matmul_and_same_scale.get_concrete_function( tensor_spec.TensorSpec( shape=input_shape, dtype=dtypes.float32, name='input_tensor' ) ), ) return model def _create_conv2d_model( self, input_shape: Sequence[int], filter_shape: Sequence[int], saved_model_path: str, bias_fn: Optional[ops.Operation] = None,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, graphdef_conversion_options, context); } return GraphdefToMlirTranslateFunction(file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0)