Home
last modified time | relevance | path

Searched refs:op_output (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_functional_ops.py125 op_output = op(*inner_args, **inner_kwargs)
128 if not op_output.shape[:1].is_compatible_with([flat_values_nrows]):
133 (op_output.shape, flat_values_nrows))
136 op_output, nested_splits_lists[0], validate=False)
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc738 std::unique_ptr<TensorT> op_output; in QuantizeOpInput() local
741 &op_output); in QuantizeOpInput()
742 op_output->quantization = absl::make_unique<QuantizationParametersT>(); in QuantizeOpInput()
743 op_output->quantization->min.push_back(tensor->quantization->min[0]); in QuantizeOpInput()
744 op_output->quantization->max.push_back(tensor->quantization->max[0]); in QuantizeOpInput()
746 op_output.get(), activations_type, error_reporter)); in QuantizeOpInput()
748 subgraph->tensors.push_back(std::move(op_output)); in QuantizeOpInput()
768 std::unique_ptr<TensorT> op_output; in QuantizeOpInput() local
770 tensor->shape_signature, TensorType_FLOAT32, &op_output); in QuantizeOpInput()
772 subgraph->tensors.push_back(std::move(op_output)); in QuantizeOpInput()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py555 def _zeros_like(op_input, op_output): argument
557 if op_output.dtype == dtypes.resource:
562 gen_resource_variable_ops.variable_shape(op_output),
564 return array_ops.zeros_like(op_output)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2_test.py1007 for op_output in outputs[op]:
1008 inputs[op_output].remove(op)
1009 if not inputs[op_output]:
1010 sources.append(op_output)
/external/llvm-project/mlir/docs/
DDeclarativeRewrites.md340 AnyType:$op_output
350 AnyType:$op_output