Home
last modified time | relevance | path

Searched refs:step_tensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc36 const auto& step_tensor = context->tensors[inputs->data[3]]; in PopulateSubGraph() local
42 graph_builder_->AddConstNodeWithData(inputs->data[3], step_tensor); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc433 const auto& step_tensor = context->tensors[node->inputs->data[3]]; in IsNodeSupportedByHexagon() local
435 !IsConstantTensor(&ends_tensor) || !IsConstantTensor(&step_tensor)) in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc989 Tensor step_tensor = CreateTensor<int64>(TensorShape({}), {step_}); in GetInputTensors() local
990 return {start_tensor, stop_tensor, step_tensor}; in GetInputTensors()