Home
last modified time | relevance | path

Searched defs:reshaped_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc72 std::vector<int64> reshaped_shape; in Compile() local
Dspacetodepth_op.cc72 std::vector<int64> reshaped_shape; in Compile() local
Dbatchtospace_op.cc70 std::vector<int64> reshaped_shape(input_rank + block_rank); in BatchToSpace() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc59 const Shape& reshaped_shape, in GetReshapedIndex()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2596 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
Dalgebraic_simplifier_test.cc4896 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local