Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc109 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in CheckOutputSize() local
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc127 int32_t dim_shape = std::ceil((end - begin) / static_cast<float>(stride)); in ResizeOutputTensor() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc234 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc237 auto* dim_shape = proto->add_dim(); in ShapeHandleToProto() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2212 const Shape& dim_shape = ShapeUtil::GetSubshape(hlo->shape(), {i}); in HandlePadToStatic() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc855 const Shape& dim_shape = TypeToShape(pad_to_static.output()[i].getType()); in EmitPadToStaticFromMlir() local