Home
last modified time | relevance | path

Searched defs:reshaped_shape (Results 1 – 11 of 11) 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/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp200 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtree_reduction_rewriter.cc146 Shape reshaped_shape = in RewriteReduction() local
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFFTConvolutionLayer.cpp207 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
DCLGEMMDeconvolutionLayer.cpp171 …const TensorShape reshaped_shape = TensorShape(nhwc_weights_shape[0], nhwc_weights_shape[1] * nhwc… in validate() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1010 auto reshaped_shape = ValuesToRank1( in matchAndRewrite() local
1115 std::vector<int64_t> reshaped_shape; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3971 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
Dalgebraic_simplifier_test.cc5969 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc605 Shape reshaped_shape = in AddBroadcastSequence() local