Home
last modified time | relevance | path

Searched refs:reshaped_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc72 std::vector<int64> reshaped_shape; in Compile() local
75 reshaped_shape.reserve(input_rank); in Compile()
79 reshaped_shape.push_back(input_shape[0]); in Compile()
81 reshaped_shape.push_back(input_shape[1 + i]); in Compile()
85 reshaped_shape.push_back(block_size_); in Compile()
88 reshaped_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
104 reshaped_shape.push_back(input_shape[0]); in Compile()
107 reshaped_shape.push_back(block_size_); in Compile()
110 reshaped_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
112 reshaped_shape.push_back(input_shape[2 + i]); in Compile()
[all …]
Dspacetodepth_op.cc72 std::vector<int64> reshaped_shape; in Compile() local
75 reshaped_shape.reserve(input_rank); in Compile()
88 reshaped_shape.push_back(input_shape[0]); in Compile()
90 reshaped_shape.push_back(input_shape[1 + i] / block_size_); in Compile()
91 reshaped_shape.push_back(block_size_); in Compile()
93 reshaped_shape.push_back(input_shape[feature_dim]); in Compile()
120 reshaped_shape.push_back(input_shape[0]); in Compile()
121 reshaped_shape.push_back(input_shape[feature_dim]); in Compile()
123 reshaped_shape.push_back(input_shape[2 + i] / block_size_); in Compile()
124 reshaped_shape.push_back(block_size_); in Compile()
[all …]
Dbatchtospace_op.cc70 std::vector<int64> reshaped_shape(input_rank + block_rank); in BatchToSpace() local
71 std::copy(block_shape.begin(), block_shape.end(), reshaped_shape.begin()); in BatchToSpace()
72 reshaped_shape[block_rank] = batch_size / block_num_elems; in BatchToSpace()
74 reshaped_shape.begin() + block_rank + 1); in BatchToSpace()
75 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in BatchToSpace()
85 std::vector<int64> permutation(reshaped_shape.size()); in BatchToSpace()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc59 const Shape& reshaped_shape, in GetReshapedIndex() argument
75 return IrArray::Index(linear_index, reshaped_shape, b); in GetReshapedIndex()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py752 reshaped_ind, reshaped_shape = gen_sparse_ops.sparse_reshape(
777 reshaped_shape = reshaped_shape_const
781 reshaped_shape)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dbeam_search_decoder.py196 reshaped_shape = tensor_shape.TensorShape([batch_size, beam_width, None])
206 % (reshaped_shape, shape))
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2596 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
2597 if (!LayoutUtil::HasLayout(reshaped_shape)) { in HandleReshape()
2598 LayoutUtil::SetToDefaultLayout(&reshaped_shape); in HandleReshape()
2601 Literal::CreateFromShape(reshaped_shape)); in HandleReshape()
Dalgebraic_simplifier_test.cc4896 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
4897 reshaped_shape.clear_layout(); in TEST_F()
4899 HloInstruction::CreateReshape(reshaped_shape, broadcast)); in TEST_F()