| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 80 static Shape RowMajorShape(const Shape& old_shape) { in RowMajorShape() argument
81 Shape new_shape(old_shape); in RowMajorShape()
88 static Shape ColMajorShape(const Shape& old_shape) { in ColMajorShape() argument
89 Shape new_shape(old_shape); in ColMajorShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 244 auto check_size_increase = [&](const Shape& old_shape, in TryResolvePaddedShapesForTensorCore()
246 int64 old_bytes = ShapeUtil::ByteSizeOf(old_shape); in TryResolvePaddedShapesForTensorCore()
254 << ShapeUtil::HumanString(old_shape) << " to " in TryResolvePaddedShapesForTensorCore()
353 auto check_size_increase = [&](const Shape& old_shape, in TryResolvePaddedShapesForIntegerConvolution()
355 int64 old_bytes = ShapeUtil::ByteSizeOf(old_shape); in TryResolvePaddedShapesForIntegerConvolution()
363 << ShapeUtil::HumanString(old_shape) << " to " in TryResolvePaddedShapesForIntegerConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier.cc | 263 const Shape& old_shape = conditional->operand(branch + 1)->shape(); in TryRemoveUnusedConditionalOperands() local
272 old_shape.tuple_shapes(i), in TryRemoveUnusedConditionalOperands()
381 const Shape old_shape = conditional_op->shape(); in RemoveUnusedTupleElements() local
386 old_shape.tuple_shapes(new_to_old_mapping[new_index])); in RemoveUnusedTupleElements()
395 old_shape)) { in RemoveUnusedTupleElements()
398 << old_shape.ToString() << ", but got " in RemoveUnusedTupleElements()
427 old_shape.tuple_shapes(old_index), old_root, old_index))); in RemoveUnusedTupleElements()
|
| D | hlo_element_type_converter.cc | 195 Shape old_shape = hlo->shape(); in Run() local
205 new_hlo = ConvertTupleElements(new_hlo, old_shape); in Run()
|
| D | conditional_code_motion.cc | 688 const Shape& old_shape = old_root->shape(); in MoveInstructionIn() local
689 for (int64 i = 0; i < old_shape.tuple_shapes_size(); ++i) { in MoveInstructionIn()
692 old_shape.tuple_shapes(i), old_root, i)); in MoveInstructionIn()
|
| D | space_to_batch_converter.cc | 1419 auto old_shape = consumer->mutable_operand(0)->shape(); in Propagate() local
1654 auto old_shape = old_instr->shape(); in SelectValidPortion() local
1660 const int64 old_batch_size = old_shape.dimensions(old_batch_dim); in SelectValidPortion()
1661 const int64 old_space_size = old_shape.dimensions(old_space_dim); in SelectValidPortion()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 114 old_shape = array_ops.shape(x)
115 new_shape = array_ops.concat([[old_shape[0] * old_shape[1]], old_shape[2:]],
|
| D | pfor.py | 1729 old_shape = array_ops.shape(x)
1730 new_shape = array_ops.concat([[-1], old_shape[2:]], axis=0)
1736 old_shape = array_ops.shape(x)
1737 new_shape = array_ops.concat([first_dim, [-1], old_shape[1:]], axis=0)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 2742 old_shape = value.shape.as_list()
2743 new_shape = old_shape[:outer_axis] + [-1] + old_shape[inner_axis + 1:]
2745 old_shape = array_ops.shape(value)
2747 [old_shape[:outer_axis], [-1], old_shape[inner_axis + 1:]], axis=0)
2768 old_shape = new_values.shape.as_list()
2769 new_shape = [-1] + old_shape[shape_split:]
2770 flat_size = _prod(old_shape[1:shape_split])
2772 old_shape = array_ops.shape(new_values)
2773 new_shape = array_ops.concat([[-1], old_shape[shape_split:]], axis=0)
2775 math_ops.reduce_prod(old_shape[1:shape_split]), new_splits.dtype)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 192 old_shape = array_ops.shape(result_t)
194 [array_ops.ones(ndmin - ndims, dtypes.int32), old_shape], axis=0)
1085 def _pad_left_to(n, old_shape): argument
1086 old_shape = asarray(old_shape, dtype=np.int32)
1088 old_shape, [[math_ops.maximum(n - array_ops.size(old_shape), 0), 0]],
1135 def new_shape(_, old_shape): argument
1137 ndim_ = array_ops.size(old_shape)
1143 old_shape, [[1, 1]], constant_values=1), lambda: array_ops.pad(
1144 old_shape, [[0, 1]], constant_values=1)))
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 342 inline std::vector<int64_t> GetPaddedShape(ArrayRef<int64_t> old_shape, in GetPaddedShape() argument
345 std::copy_backward(old_shape.begin(), old_shape.end(), new_shape.end()); in GetPaddedShape()
1290 auto old_shape = shape.getType().getShape(); in matchAndRewrite() local
1291 if (old_shape.size() == 1) { in matchAndRewrite()
1295 for (auto it = ++old_shape.rbegin(); it != old_shape.rend(); ++it) { in matchAndRewrite()
1303 {*old_shape.rbegin()}, shape.getType().getElementType())); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 4539 old_shape = array_ops.shape(x)
4540 new_shape = array_ops.stack([old_shape[0], s], axis=0)
|