| /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 | 106 TfLiteIntArray* new_shape) { in WriteToTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores.cc | 46 const Shape& new_shape) { in PadInstruction()
185 const Shape& new_shape) { in PadForTensorCores()
|
| D | variadic_op_splitter.cc | 58 Shape new_shape = concat->shape(); in SplitConcatenate() local
|
| D | cudnn_conv_rewriter.cc | 472 Shape new_shape = rhs->shape(); in MatchBackwardInput() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 139 std::vector<int64> new_shape(existing_dims_size); in Compile() local
172 std::vector<int64> new_shape; in Compile() local
|
| D | dynamic_stitch_op.cc | 154 TensorShape new_shape; in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 420 def _reshape_if_necessary(tensor, new_shape): argument
|
| D | sparse_ops.py | 1710 def sparse_reset_shape(sp_input, new_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter.cc | 196 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, in Run() local
|
| D | reshape_mover.cc | 142 const Shape new_shape = in UpdateOperand() local
|
| D | hlo_creation_utils.cc | 116 Shape new_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in MakeReshapeHlo() local
358 Shape new_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in ExpandFirstDimIntoNDims() local
|
| D | scatter_expander.cc | 64 auto new_shape = in CanonicalizeScatterIndices() 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/core/framework/ |
| D | shape_inference.h | 247 ShapeHandle new_shape; in MergeInput() local
279 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/contrib/distributions/python/ops/ |
| D | batch_reshape.py | 373 def calculate_reshape(original_shape, new_shape, validate=False, name=None): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 108 TensorShape new_shape = TensorShape({old_shape_.num_elements()}); in Flatten() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 371 ShapeHandle new_shape; in __anona7ffb82a1002() local
|
| D | array_ops_test.cc | 871 Tensor new_shape = test::AsTensor<int32>({1, 2, 3}); in TEST() local
926 Tensor new_shape = test::AsTensor<int32>({1, 2, 3}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 411 IrArray IrArray::CastToShape(const Shape& new_shape, in CastToShape()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 459 auto* new_shape = schema_params->new_shape(); in ParseOpData() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 163 auto new_shape = ops::Const(root.WithOpName("new_shape"), {3, -1}, {2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 764 Shape new_shape = original; in ChangeElementType() local
932 Shape new_shape = shape; in PermuteDimensions() local
|