Home
last modified time | relevance | path

Searched refs:bound_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc78 const TensorShape bound_shape = ctx->InputShape("bound"); in Compile() local
95 ctx, bound_shape.dims() == 0, in Compile()
98 bound_shape.DebugString())); in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc554 auto bound_shape = dest_subshape.is_static() ? src_subshape : dest_subshape; in CopyFrom() local
557 CHECK(ShapeUtil::DynamicShapeIsCompatible(compact_shape, bound_shape)) in CopyFrom()
558 << compact_shape.ToString() << " vs " << bound_shape.ToString(); in CopyFrom()
1573 auto bound_shape = dest_shape.is_static() ? src_shape : dest_shape; in CopyElementsWithDynamicBound() local
1593 } while (IndexUtil::BumpIndices(bound_shape, absl::MakeSpan(index))); in CopyElementsWithDynamicBound()