Searched refs:input_dtypes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 478 input_dtypes = {}
482 input_dtypes[self._ToString(node.name)] = (
484 assert len(params.input_names) == len(input_dtypes)
490 dtype = input_dtypes[params.input_names[i]]
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 64 input_dtypes = [seq_inputs.dtype] + _GetDTypesFromStructure(initial_state)
70 @function.Defun(*input_dtypes)
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 132 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
209 gtl::ArraySlice<tensorflow::DataType> input_dtypes, in RecordOperation() argument
212 if (!ShouldRecord(input_tensor_id, input_dtypes)) { in RecordOperation()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 317 std::vector<DataType> input_dtypes; in BuildXlaHostComputeNodeDef() local
318 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->attrs(), "Tinputs", &input_dtypes)); in BuildXlaHostComputeNodeDef()
319 std::vector<NodeDefBuilder::NodeOut> inputs(input_dtypes.size()); in BuildXlaHostComputeNodeDef()
325 if (e->dst_input() < 0 || e->dst_input() >= input_dtypes.size()) { in BuildXlaHostComputeNodeDef()
329 e->src()->name(), e->src_output(), input_dtypes[e->dst_input()]}; in BuildXlaHostComputeNodeDef()
|
| D | encapsulate_subgraphs_pass.cc | 949 std::vector<DataType> input_dtypes(oc_subgraph.inputs.size(), DT_INVALID); in AddHostComputes() local
961 input_dtypes[input_index] = dtype; in AddHostComputes()
985 builder.Attr("Tinputs", input_dtypes); in AddHostComputes()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1578 const std::vector<tensorflow::DataType>& input_dtypes, argument
1615 input_dtypes, backward_function_getter,
1630 std::vector<tensorflow::DataType> input_dtypes = local
1635 op_type, output_tensors, input_ids, input_dtypes,
2078 std::vector<tensorflow::DataType> input_dtypes = MakeTensorDtypeList(inputs); local
2083 if (tape->tape->ShouldRecord(input_ids, input_dtypes)) {
2127 op_name, results, input_ids, input_dtypes,
|