Home
last modified time | relevance | path

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

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