| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 66 std::vector<xla::XlaOp> input_handles; in Compile() local
69 ctx->InputList("inputs", &input_handles, &input_shapes)); in Compile()
73 ctx, input_handles.size() == 1, in Compile()
75 "Einsum Op has ", input_handles.size(), " inputs ", in Compile()
78 ctx->SetOutput(0, xla::Einsum(input_handles[0], equation_)); in Compile()
81 ctx, input_handles.size() == 2, in Compile()
83 "Einsum Op has ", input_handles.size(), " inputs ", in Compile()
87 0, xla::Einsum(input_handles[0], input_handles[1], equation_)); in Compile()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 213 jlongArray input_handles = env->NewLongArray(ninputs); in buildSubgraph() local
219 env->GetLongArrayElements(input_handles, nullptr); in buildSubgraph()
236 env->ReleaseLongArrayElements(input_handles, input_handles_elems, 0); in buildSubgraph()
245 input_handles, input_indices, output_handles, output_indices); in buildSubgraph()
257 JNIEnv* env, jclass clazz, jlong handle, jlongArray input_handles, in Java_org_tensorflow_Graph_whileLoop() argument
264 int ninputs = env->GetArrayLength(input_handles); in Java_org_tensorflow_Graph_whileLoop()
267 resolveOutputs(env, "inputs", input_handles, input_indices, inputs.get(), in Java_org_tensorflow_Graph_whileLoop()
|
| D | eager_operation_builder_jni.cc | 141 JNIEnv* env, jclass clazz, jlong op_handle, jlongArray input_handles) { in Java_org_tensorflow_EagerOperationBuilder_addInputList() argument
144 jlong* cinput_handles = env->GetLongArrayElements(input_handles, nullptr); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
145 size_t num_inputs = static_cast<size_t>(env->GetArrayLength(input_handles)); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
151 env->ReleaseLongArrayElements(input_handles, cinput_handles, JNI_ABORT); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
155 env->ReleaseLongArrayElements(input_handles, cinput_handles, JNI_ABORT); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TensorListPushBackBatch.pbtxt | 4 name: "input_handles"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TensorListPushBackBatch.pbtxt | 4 name: "input_handles"
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 133 std::vector<xla::XlaOp> input_handles; in Compile() local
135 auto inputs = ctx->InputList("inputs", &input_handles, &input_shapes); in Compile()
149 for (int i = 0; i < input_handles.size(); ++i) { in Compile()
166 xla::SendToHost(input_handles[i], token, xla_shape, channel)); in Compile()
172 if (!input_handles.empty()) { in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | list_ops.cc | 119 shape_inference::ShapeHandle input_handles; in __anonabedecb60402() local
120 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &input_handles)); in __anonabedecb60402()
126 c->MergePrefix(tensor, input_handles, &tensor, &input_handles)); in __anonabedecb60402()
128 c->set_output(0, input_handles); in __anonabedecb60402()
|
| D | ops.pbtxt | 55647 name: "input_handles"
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 331 std::vector<const TensorHandle*> input_handles; in InputTFE_OutputTensorHandles() local
334 input_handles.push_back(TensorHandleFromInterface(abstract_tensor_handle)); in InputTFE_OutputTensorHandles()
357 *selected_device, context, input_handles); in InputTFE_OutputTensorHandles()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 5033 …argspec: "args=[\'input_handles\', \'tensor\', \'name\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 5033 …argspec: "args=[\'input_handles\', \'tensor\', \'name\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 83596 name: "input_handles"
|