Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_parameterized_test.py82 expected_inputs = [
89 expected_inputs.append(conv_scope + delim + 'depthwise/ReadVariableOp')
91 expected_inputs.append(conv_scope + delim + layer + '/ReadVariableOp')
93 expected_inputs = [
98 expected_inputs.append(conv_scope + delim + 'depthwise_weights/read')
100 expected_inputs.append(conv_scope + delim + 'weights/read')
102 self._AssertInputOpsAre(weights_quant, expected_inputs)
119 expected_inputs = [
127 expected_inputs = [
132 self._AssertInputOpsAre(conv_quant, expected_inputs)
[all …]
Dquantize_graph_test.py523 expected_inputs = [in_op_name + ':0' for in_op_name in in_op_names]
524 self.assertItemsEqual([t.name for t in op.inputs], expected_inputs)
Dfold_batch_norms_test.py779 expected_inputs = [in_op_name + ':0' for in_op_name in in_op_names]
780 self.assertItemsEqual([t.name for t in op.inputs], expected_inputs)
/external/tensorflow/tensorflow/core/kernels/
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.cc76 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64, DT_INT64}; in CheckSignature() local
77 expected_inputs.push_back(accumulator->dtype()); in CheckSignature()
78 expected_inputs.push_back(DT_INT64); in CheckSignature()
79 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
Dconditional_accumulator_op.cc76 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64}; in CheckSignature() local
77 expected_inputs.push_back(accumulator->dtype()); in CheckSignature()
78 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
Dlookup_table_init_op.cc55 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
58 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
123 DataTypeVector expected_inputs = {expected_input_0, DT_STRING}; in Compute() local
125 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
Dconditional_accumulator_base_op.h94 DataTypeVector expected_inputs = {}; in SetAccumulatorHandle() local
95 TF_RETURN_IF_ERROR(ctx->MatchSignature(expected_inputs, {DT_STRING_REF})); in SetAccumulatorHandle()
Dlookup_table_op.cc794 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
797 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
830 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
832 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
867 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype()}; in Compute() local
868 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
940 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute() local
942 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()
/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.cc57 Status MatchSignatureHelper(const DataTypeSlice expected_inputs, in MatchSignatureHelper() argument
63 if (inputs.size() != expected_inputs.size()) signature_mismatch = true; in MatchSignatureHelper()
65 if (!TypesCompatible(expected_inputs[i], inputs[i])) { in MatchSignatureHelper()
81 " expected: ", DataTypeSliceString(expected_inputs), "->", in MatchSignatureHelper()
230 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature() argument
231 return MatchSignatureHelper(expected_inputs, expected_outputs, input_types_, in MatchSignature()
872 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature() argument
879 return MatchSignatureHelper(expected_inputs, expected_outputs, inputs, in MatchSignature()
Dop_kernel.h331 Status MatchSignature(const DataTypeSlice expected_inputs,
1131 Status MatchSignature(const DataTypeSlice expected_inputs,
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc870 std::vector<string> expected_inputs, test_inputs; in ExpectTfAndXlaOutputsAreClose() local
874 /*use_jit=*/false, &graph, /*test_node_def=*/nullptr, &expected_inputs, in ExpectTfAndXlaOutputsAreClose()
905 std::vector<std::pair<string, Tensor>> expected_feeds(expected_inputs.size()); in ExpectTfAndXlaOutputsAreClose()
907 CHECK_EQ(input_tensors.size(), expected_inputs.size()); in ExpectTfAndXlaOutputsAreClose()
911 expected_feeds[i] = {expected_inputs[i], input_tensors[i]}; in ExpectTfAndXlaOutputsAreClose()