Searched refs:leading_dim_is_dynamic (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 206 bool* leading_dim_is_dynamic, in GetLeadingDimForTensorList() argument
213 *leading_dim_is_dynamic = buffer_shape.is_dynamic_dimension(0); in GetLeadingDimForTensorList()
218 *leading_dim_is_dynamic = list_shape.is_dynamic_dimension(0); in GetLeadingDimForTensorList()
227 bool leading_dim_is_dynamic, xla::Shape* tensor_list_shape) { in GetTensorListShapeFromElementTensorListShape() argument
237 if (leading_dim_is_dynamic) { in GetTensorListShapeFromElementTensorListShape()
249 bool leading_dim_is_dynamic, in GetTensorListShapeFromElementShape() argument
262 shapes.back().set_dynamic_dimension(0, leading_dim_is_dynamic); in GetTensorListShapeFromElementShape()
301 bool leading_dim_is_dynamic; in GetInitializedTensorListForElement() local
303 list, &leading_dim, &leading_dim_is_dynamic, &leading_dim_dynamic_size)); in GetInitializedTensorListForElement()
311 element_shape, leading_dim, leading_dim_is_dynamic, &list_shape)); in GetInitializedTensorListForElement()
[all …]
|
| D | tensor_list_utils.h | 71 bool* leading_dim_is_dynamic,
78 bool leading_dim_is_dynamic,
|
| D | tensor_list_ops.cc | 60 bool leading_dim_is_dynamic; in GetTensorListDynamicDims() local
62 ctx->ResolveInputDynamismIntoPred(1, &leading_dim_is_dynamic)); in GetTensorListDynamicDims()
66 if (leading_dim_is_dynamic) { in GetTensorListDynamicDims()
94 bool leading_dim_is_dynamic; in Compile() local
96 &leading_dim_is_dynamic, in Compile()
|