Home
last modified time | relevance | path

Searched refs:expected_inputs (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator_op.cc137 DataTypeVector expected_inputs; in GetExpectedInputs() local
138 expected_inputs = {DT_STRING_REF, DT_INT64}; in GetExpectedInputs()
139 expected_inputs.push_back(accumulator->dtype()); in GetExpectedInputs()
140 return expected_inputs; in GetExpectedInputs()
158 DataTypeVector expected_inputs; in GetExpectedInputs() local
159 expected_inputs = {DT_RESOURCE, DT_INT64}; in GetExpectedInputs()
160 expected_inputs.push_back(accumulator->dtype()); in GetExpectedInputs()
161 return expected_inputs; in GetExpectedInputs()
Dqueue_op.cc90 DataTypeVector expected_inputs; in ComputeAsync() local
92 expected_inputs.push_back(DT_RESOURCE); in ComputeAsync()
94 expected_inputs.push_back(DT_STRING_REF); in ComputeAsync()
97 expected_inputs.push_back(dt); in ComputeAsync()
99 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync()
131 DataTypeVector expected_inputs; in ComputeAsync() local
133 expected_inputs.push_back(DT_RESOURCE); in ComputeAsync()
135 expected_inputs.push_back(DT_STRING_REF); in ComputeAsync()
138 expected_inputs.push_back(dt); in ComputeAsync()
140 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync()
Dsparse_conditional_accumulator_op.cc87 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64, DT_INT64}; in GetExpectedInputs() local
88 expected_inputs.push_back(accumulator->dtype()); in GetExpectedInputs()
89 expected_inputs.push_back(DT_INT64); in GetExpectedInputs()
90 return expected_inputs; in GetExpectedInputs()
Dlookup_table_init_op.cc56 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
59 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
125 DataTypeVector expected_inputs = {expected_input_0, DT_STRING}; in Compute() local
127 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
Dconditional_accumulator_base_op.h151 DataTypeVector expected_inputs = GetExpectedInputs(accumulator); in CheckSignature() local
152 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
Dlookup_table_op.cc834 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
837 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
868 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
870 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
903 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype()}; in Compute() local
904 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
974 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute() local
976 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
Dbarrier_ops.cc591 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT32}; in ComputeAsync() local
598 ctx, ctx->MatchSignature(expected_inputs, expected_outputs), callback); in ComputeAsync()
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc277 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; in TEST_F() local
280 BuildSignatureDef(expected_inputs, expected_outputs); in TEST_F()
288 EXPECT_EQ(expected_inputs, inputs); in TEST_F()
307 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; in TEST_F() local
309 EXPECT_EQ(expected_inputs, inputs); in TEST_F()
449 std::unordered_set<string> expected_inputs = {"input1:0", "input2:0"}; in TEST_F() local
451 EXPECT_EQ(expected_inputs, inputs); in TEST_F()
478 std::unordered_set<string> expected_inputs = {"input1:0", "input2:0", in TEST_F() local
482 EXPECT_EQ(expected_inputs, inputs); in TEST_F()
/external/grpc-grpc/src/ruby/spec/generic/
Dclient_stub_spec.rb984 def run_bidi_streamer_handle_inputs_first(expected_inputs, replys, argument
988 expected_inputs.each { |i| expect(c.remote_read).to eq(i) }
995 def run_bidi_streamer_echo_ping_pong(expected_inputs, status, client_starts, argument
1006 expected_inputs.each do |i|
1021 def run_client_streamer(expected_inputs, resp, status, argument
1029 expected_inputs.each { |i| expect(c.remote_read).to eq(i) }
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc63 Status MatchSignatureHelper(const DataTypeSlice expected_inputs, in MatchSignatureHelper() argument
69 if (inputs.size() != expected_inputs.size()) signature_mismatch = true; in MatchSignatureHelper()
71 if (!TypesCompatible(expected_inputs[i], inputs[i])) { in MatchSignatureHelper()
87 " expected: ", DataTypeSliceString(expected_inputs), "->", in MatchSignatureHelper()
255 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature() argument
256 return MatchSignatureHelper(expected_inputs, expected_outputs, in MatchSignature()
1031 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature() argument
1038 return MatchSignatureHelper(expected_inputs, expected_outputs, inputs, in MatchSignature()
Dop_kernel.h327 Status MatchSignature(const DataTypeSlice expected_inputs,
1168 Status MatchSignature(const DataTypeSlice expected_inputs,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2052 const std::vector<std::pair<string, TrtInputArg>>& expected_inputs) { in CheckInputsWeights() argument
2055 if (inputs.size() != expected_inputs.size()) { in CheckInputsWeights()
2058 expected_inputs.size(), ", at ", node_def.name()); in CheckInputsWeights()
2061 if (expected_inputs[i].second == TrtInputArg::kWeight && in CheckInputsWeights()
2063 return errors::Unimplemented("The input \"", expected_inputs[i].first, in CheckInputsWeights()
2071 if (expected_inputs[i].second == TrtInputArg::kTensor && in CheckInputsWeights()
2073 return errors::Unimplemented("The input \"", expected_inputs[i].first, in CheckInputsWeights()
2087 std::vector<std::pair<string, TrtInputArg>> expected_inputs; in CheckInputsWeights() local
2088 expected_inputs.reserve(inputs_is_weight.size()); in CheckInputsWeights()
2091 std::back_inserter(expected_inputs), [](std::pair<string, bool> x) { in CheckInputsWeights()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc871 std::vector<string> expected_inputs, test_inputs; in ExpectTfAndXlaOutputsAreClose() local
875 /*use_jit=*/false, &graph, /*test_node_def=*/nullptr, &expected_inputs, in ExpectTfAndXlaOutputsAreClose()
906 std::vector<std::pair<string, Tensor>> expected_feeds(expected_inputs.size()); in ExpectTfAndXlaOutputsAreClose()
908 CHECK_EQ(input_tensors.size(), expected_inputs.size()); in ExpectTfAndXlaOutputsAreClose()
912 expected_feeds[i] = {expected_inputs[i], input_tensors[i]}; in ExpectTfAndXlaOutputsAreClose()