| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 81 Shape new_shape(old_shape); in RowMajorShape() local
89 Shape new_shape(old_shape); in ColMajorShape() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | string_util_test.cc | 57 auto new_shape = TfLiteIntArrayCreate(2); in TEST() local
147 auto new_shape = TfLiteIntArrayCreate(2); in TEST() local
|
| D | string_util.cc | 99 TfLiteIntArray* new_shape) { in WriteToTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 46 const Shape& new_shape) { in PadInstruction()
245 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()
354 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
|
| D | variadic_op_splitter.cc | 58 Shape new_shape = concat->shape(); in SplitConcatenate() local
|
| D | gpu_conv_rewriter.cc | 306 Shape new_shape = lhs_transpose->shape(); in MatchBackwardFilter() local
608 Shape new_shape = rhs->shape(); in MatchBackwardInput() local
|
| D | cublas_gemm_pad_for_tensor_cores.cc | 67 auto create_padding_config = [](Shape& shape, Shape& new_shape) { in PadForTensorCores()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 176 std::vector<int64> new_shape(existing_dims_size); in Compile() local
209 std::vector<int64> new_shape; in Compile() local
|
| D | dynamic_stitch_op.cc | 154 TensorShape new_shape; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | slice_sinker.cc | 171 Shape new_shape = ShapeUtil::ChangeElementType(shape, element_type); in SinkSlices() local
|
| D | hlo_element_type_converter.cc | 196 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, in Run() local
|
| D | depthwise_convolution_converter.cc | 163 Shape new_shape = lhs->shape(); in HandleBackwardFilterBatchGroupConvolution() local
|
| D | reshape_mover.cc | 142 const Shape new_shape = in UpdateOperand() local
|
| D | scatter_expander.cc | 64 auto new_shape = in CanonicalizeScatterIndices() local
|
| D | hlo_creation_utils.cc | 117 Shape new_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in MakeReshapeHlo() local
427 Shape new_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in ExpandFirstDimIntoNDims() local
|
| D | conditional_simplifier.cc | 354 const Shape new_shape = ShapeUtil::MakeTupleShape(new_tuple_shapes); in RemoveUnusedTupleElements() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | cwise_op_vectorizer.cc | 100 Output new_shape = ops::Concat(scope, {concat_pre, ones, concat_post}, in ExpandDimsForBroadcast() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 366 HW new_shape; member
377 HWD new_shape; member
453 BHWC new_shape; member
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_nearest_upsample.cc | 52 void CopyShape(const std::vector<int>& new_shape, Array* array) { in CopyShape()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 215 ShapeHandle new_shape; in MergeInput() local
247 ShapeHandle new_shape; in RelaxInput() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 202 std::vector<int64> new_shape; in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 684 def _reshape_if_necessary(tensor, new_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 283 Shape new_shape = shape; in MakeFakeLiteralInternal() local
385 Shape new_shape = shape; in MakeFakeLiteralInternalWithBounds() local
|
| /external/tensorflow/tensorflow/c/ |
| D | ops_test.cc | 110 TF_ShapeHandle* new_shape = TF_ShapeInferenceContextVectorFromSize( in vectorize_shape_fn() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 110 TensorShape new_shape = TensorShape({old_shape_.num_elements()}); in Flatten() local
|