Home
last modified time | relevance | path

Searched defs:expected_inputs (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/
Dsimple_hash_table_kernel.cc176 DataTypeVector expected_inputs = {DT_RESOURCE, DataTypeToEnum<K>::v(), in Compute() local
199 DataTypeVector expected_inputs = {DT_RESOURCE, DataTypeToEnum<K>::v(), in Compute() local
218 DataTypeVector expected_inputs = {DT_RESOURCE, DataTypeToEnum<K>::v()}; in Compute() local
235 DataTypeVector expected_inputs = {DT_RESOURCE}; in Compute() local
254 DataTypeVector expected_inputs = {DT_RESOURCE, DataTypeToEnum<K>::v(), in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_init_op.cc59 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
130 DataTypeVector expected_inputs = {expected_input_0, DT_STRING}; in Compute() local
Dconditional_accumulator_op.cc137 DataTypeVector expected_inputs; in GetExpectedInputs() local
158 DataTypeVector expected_inputs; in GetExpectedInputs() local
Dqueue_op.cc90 DataTypeVector expected_inputs; in ComputeAsync() local
131 DataTypeVector expected_inputs; in ComputeAsync() local
Dsparse_conditional_accumulator_op.cc87 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64, DT_INT64}; in GetExpectedInputs() local
Dconditional_accumulator_base_op.h150 DataTypeVector expected_inputs = GetExpectedInputs(accumulator); in CheckSignature() local
Dlookup_table_op.cc914 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
948 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
983 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype()}; in Compute() local
1054 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
Dbarrier_ops.cc587 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT32}; in ComputeAsync() local
/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_4/
Dmultiplex_4_kernel.cc49 const int64_t expected_inputs = 2 * num_cond_a_ + 1; in Compute() local
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc287 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; in TEST_F() local
317 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; in TEST_F() local
463 std::unordered_set<string> expected_inputs = {"input1:0", "input2:0"}; in TEST_F() local
492 std::unordered_set<string> expected_inputs = {"input1:0", "input2:0", in TEST_F() local
/external/grpc-grpc/src/ruby/spec/generic/
Dclient_stub_spec.rb984 def run_bidi_streamer_handle_inputs_first(expected_inputs, replys, argument
995 def run_bidi_streamer_echo_ping_pong(expected_inputs, status, client_starts, argument
1021 def run_client_streamer(expected_inputs, resp, status, argument
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel.h74 Status MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc69 Status MatchSignatureHelper(const DataTypeSlice expected_inputs, in MatchSignatureHelper()
284 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature()
1042 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1667 const std::vector<std::pair<string, TrtInputArg>>& expected_inputs) { in CheckInputsWeights()
1704 std::vector<std::pair<string, TrtInputArg>> expected_inputs; in CheckInputsWeights() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1492 std::vector<string> expected_inputs, test_inputs; in ExpectTfAndXlaOutputsAreClose() local