Searched refs:original_input (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | lift_to_graph.py | 88 for input_index, original_input in enumerate(op.inputs):
89 copied_input = op_map.get(original_input, None)
96 shape=original_input.shape,
97 dtype=original_input.dtype)
102 old_graph_tensor=original_input))
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | UnbatchGrad.pbtxt | 4 name: "original_input"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | UnbatchGrad.pbtxt | 4 name: "original_input"
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 70 const MaybeOwningDeviceMemory& original_input = in AllocateOutputMemoryWithInputReuse() local
73 if (!original_input.HasOwnership()) { in AllocateOutputMemoryWithInputReuse()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_UnbatchGrad.pbtxt | 9 original_input: The input to the Unbatch operation this is the gradient of.
|
| /external/rust/crates/url/src/ |
| D | parser.rs | 208 original_input: &'i str, in trim_tab_and_newlines()
211 let input = original_input.trim_matches(ascii_tab_or_new_line); in trim_tab_and_newlines()
213 if input.len() < original_input.len() { in trim_tab_and_newlines()
225 pub fn with_log(original_input: &'i str, vfn: Option<&dyn Fn(SyntaxViolation)>) -> Self { in with_log()
226 let input = original_input.trim_matches(c0_control_or_space); in with_log()
228 if input.len() < original_input.len() { in with_log()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 655 original_input = endpoints[key]
657 self.assertEqual(original_input.name, converted_input.name)
658 self.assertEqual(original_input.dtype, converted_input.dtype)
660 tensor_shape.TensorShape(original_input.tensor_shape).as_list(),
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 531 Value original_input = strided_slice_op.input(); in RewriteNewAxisMask() local
533 original_input.getType().dyn_cast<RankedTensorType>(); in RewriteNewAxisMask()
569 loc, revised_output_type, original_input, shape); in RewriteNewAxisMask()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 1305 HloInstruction* original_input = hlo->mutable_operand(operand_index); in HandleConditional() local
1311 TF_RET_CHECK(original_input->shape().IsTuple()); in HandleConditional()
1313 new_operand = TupleUtil::AppendSuffix(original_input, operands_to_add); in HandleConditional()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 5185 …argspec: "args=[\'original_input\', \'batch_index\', \'grad\', \'id\', \'container\', \'shared_nam…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 5185 …argspec: "args=[\'original_input\', \'batch_index\', \'grad\', \'id\', \'container\', \'shared_nam…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3223 func UnbatchGrad(scope *Scope, original_input tf.Output, batch_index tf.Output, grad tf.Output, id …
3234 original_input, batch_index, grad, id,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 57024 name: "original_input"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 85111 name: "original_input"
|