Searched refs:set_dimensions (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 109 expanded_shape.set_dimensions( in ExpandedFilterShape()
256 reduce_window_shape.set_dimensions(output_batch_dimension, 1); in HandleBatchGroupCount()
411 reshaped_activation_shape.set_dimensions(activation_input_feature_dim, in HandleConvolution()
501 filter_sliced_shape.set_dimensions(kernel_output_feature_dim, in HandleConvolution()
511 activation_sliced_shape.set_dimensions(activation_input_feature_dim, in HandleConvolution()
518 conv_slice_shape.set_dimensions(output_feature_dim, filter_slice_width); in HandleConvolution()
|
| D | hlo_sharding.cc | 440 result_shape.set_dimensions( in TileShape()
|
| D | shape_inference.cc | 853 output_shape.set_dimensions(dimension_to_match, small_dimension_size); in InferInDimBroadcastShape()
1863 result.set_dimensions(result.dimensions_size() - 1, in InferFftShape()
1893 result.set_dimensions(result.dimensions_size() - 1, in InferFftShape()
|
| D | algebraic_simplifier.cc | 713 new_slice_shape.set_dimensions( in HandleConcatenate()
1458 rhs_slice_shape.set_dimensions(rhs_contracting_dim, sub_k); in OptimizeDotOfConcatHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores.cc | 171 new_input_shape->set_dimensions(dnums.input_feature_dimension(), 4); in PadForTensorCores()
172 new_filter_shape->set_dimensions(dnums.kernel_input_feature_dimension(), 4); in PadForTensorCores()
175 s->set_dimensions(dim, RoundUpToNearest<int64>(s->dimensions(dim), 8)); in PadForTensorCores()
|
| D | variadic_op_splitter.cc | 66 new_shape.set_dimensions(concat->concatenate_dimension(), in SplitConcatenate()
|
| D | cudnn_conv_rewriter.cc | 482 new_shape.set_dimensions(input_feature_dimension, in MatchBackwardInput()
484 new_shape.set_dimensions(output_feature_dimension, in MatchBackwardInput()
|
| D | cudnn_conv_padding_legalization.cc | 317 new_backward_conv_shape.set_dimensions( in CanonicalizeBackwardInputConvolution()
|
| D | ir_emitter_unnested.cc | 1276 standard_iteration_shape.set_dimensions(dimension_to_sort, in HandleSort()
1290 iteration_shape.set_dimensions(dimension_to_sort, num_iterations_in_sort_dim); in HandleSort()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 50 expanded_shape.set_dimensions( in ExpandedFilterShapeForDepthwiseConvolution()
137 implicit_broadcast_filter_shape.set_dimensions(input_feature_dim, 1); in ReshapeFilterForDepthwiseConvolution()
138 implicit_broadcast_filter_shape.set_dimensions( in ReshapeFilterForDepthwiseConvolution()
|
| D | image_resize_ops.cc | 218 broadcast_shape.set_dimensions(spatial_dimensions_offset + i, in BroadcastSpatialDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.h | 98 void set_dimensions(int index, int64 value) { dimensions_.at(index) = value; } in set_dimensions() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 280 tensor_.set_dimensions(0, value);
284 tensor_.set_dimensions(1, value);
400 tensor_.set_dimensions(0, value);
404 tensor_.set_dimensions(1, value);
|
| /external/libavc/test/encoder/ |
| D | main.c | 1009 void set_dimensions(app_ctxt_t *ps_app_ctxt, in set_dimensions() function
2482 set_dimensions(&s_app_ctxt, 0, 0); in h264enc_main()
|