Searched refs:sizes_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | encode_proto_op.cc | 525 const Tensor* sizes_tensor; in Compute() local
526 OP_REQUIRES_OK(ctx, ctx->input("sizes", &sizes_tensor)); in Compute()
579 OP_REQUIRES(ctx, sizes_tensor->shape() == expected_sizes_shape, in Compute()
582 sizes_tensor->shape().DebugString(), in Compute()
585 auto sizes = sizes_tensor->flat_inner_dims<int32>(); in Compute()
|
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | utils.cc | 416 const auto& sizes_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
417 if (!IsConstantTensor(&begins_tensor) || !IsConstantTensor(&sizes_tensor)) in IsNodeSupportedByHexagon()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops.cpp | 2709 auto sizes_tensor = torch::empty( in __anonbfe5918f9502() local
2711 auto accessor = sizes_tensor.accessor<int64_t, 1>(); in __anonbfe5918f9502()
2715 stack.emplace_back(sizes_tensor); in __anonbfe5918f9502()
|