Home
last modified time | relevance | path

Searched defs:values_tensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather.cc41 Tensor* values_tensor) { in SparsifyWeights()
351 Tensor values_tensor; in SparsifyGatherInternal() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc452 const TfLiteTensor& values_tensor, in SetOutputT()
527 const TfLiteTensor& values_tensor, in SetOutput()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dhistogram_op.cc93 const Tensor& values_tensor = ctx->input(0); in Compute() local
H A Dragged_tensor_to_tensor_op.cc481 const auto& values_tensor = context->input(kValueInputIndex); in SetOutput() local
H A Dmap_stage_op.cc531 OpInputList values_tensor; in Compute() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dunpickler.cpp903 auto& values_tensor = elements.at(idx++).toTensor(); in rebuildSparseTensor() local
917 auto& values_tensor = elements.at(idx++).toTensor(); in rebuildSparseTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
H A Dmultiplex_3_kernel.cc179 Status ValidateSparseTensor(const ::tensorflow::Tensor& indices_tensor, in ValidateSparseTensor()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc435 std::unique_ptr<parallel_device::ParallelTensor> values_tensor, in Wrap()