Searched refs:sizes_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | encode_proto_op.cc | 527 const Tensor* sizes_tensor; in Compute() local
528 OP_REQUIRES_OK(ctx, ctx->input("sizes", &sizes_tensor)); in Compute()
581 OP_REQUIRES(ctx, sizes_tensor->shape() == expected_sizes_shape, in Compute()
584 sizes_tensor->shape().DebugString(), in Compute()
587 auto sizes = sizes_tensor->flat_inner_dims<int32>(); in Compute()
|
| D | decode_proto_op.cc | 739 Tensor* sizes_tensor = nullptr; in Compute() local
740 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, sizes_shape, &sizes_tensor)); in Compute()
780 CountFields(ctx, mi, *bufs[mi], sizes_tensor, &max_sizes); in Compute()
835 Tensor* sizes_tensor, std::vector<int32>* max_sizes) { in CountFields() argument
869 auto sizes = sizes_tensor->flat_inner_dims<int32>(); in CountFields()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.cc | 407 const auto& sizes_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
408 if (!IsConstantTensor(&begins_tensor) || !IsConstantTensor(&sizes_tensor)) in IsNodeSupportedByHexagon()
|