Searched refs:leading_dim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 205 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim, in GetLeadingDimForTensorList() argument
215 *leading_dim = buffer_shape.dimensions(0); in GetLeadingDimForTensorList()
219 *leading_dim = list_shape.dimensions(0); in GetLeadingDimForTensorList()
226 const xla::Shape& element_tensor_list_shape, int64 leading_dim, in GetTensorListShapeFromElementTensorListShape() argument
234 dimensions.insert(dimensions.begin(), leading_dim); in GetTensorListShapeFromElementTensorListShape()
248 int64 leading_dim, in GetTensorListShapeFromElementShape() argument
259 dimensions.insert(dimensions.begin(), leading_dim); in GetTensorListShapeFromElementShape()
299 int64 leading_dim; 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 | 70 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim,
77 int64 leading_dim,
|
| D | tensor_list_ops.cc | 92 int64 leading_dim; in Compile() local
95 OP_REQUIRES_OK(ctx, GetLeadingDimForTensorList(ctx->Input(0), &leading_dim, in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 441 Value leading_dim = GetNumElements(op, operands, &rewriter); in matchAndRewrite() local
447 if (matchPattern(leading_dim, m_Constant(&element_attr))) { in matchAndRewrite()
468 ArrayRef<Value>({leading_dim, element_shape})); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 388 int64 leading_dim = 0; in Compute() local
414 leading_dim += dim; in Compute()
417 output_shape.InsertDim(0, leading_dim); in Compute()
|