| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| quantize_parameterized_test.py | 96 expected_inputs = [
100 self._AssertInputOpsAre(weights_quant, expected_inputs)
112 expected_inputs = [
116 self._AssertInputOpsAre(conv_quant, expected_inputs)
125 expected_inputs = [
129 self._AssertInputOpsAre(act_quant, expected_inputs)
173 expected_inputs = [
177 self._AssertInputOpsAre(weights_quant, expected_inputs)
188 expected_inputs = [
192 self._AssertInputOpsAre(conv_quant, expected_inputs)
[all...] |
| fold_batch_norms_test.py | 450 expected_inputs = [in_op_name + ':0' for in_op_name in in_op_names]
451 self.assertItemsEqual([t.name for t in op.inputs], expected_inputs)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_conditional_accumulator_op.cc | 76 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64, DT_INT64}; variable
77 expected_inputs.push_back(accumulator->dtype());
78 expected_inputs.push_back(DT_INT64);
79 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {}));
|
| conditional_accumulator_op.cc | 75 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64}; variable
76 expected_inputs.push_back(accumulator->dtype());
77 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {}));
|
| lookup_table_init_op.cc | 55 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), variable
58 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
125 DataTypeVector expected_inputs = {expected_input_0, DT_STRING}; variable
127 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
|
| queue_ops.cc | 85 DataTypeVector expected_inputs; variable
87 expected_inputs.push_back(DT_RESOURCE);
89 expected_inputs.push_back(DT_STRING_REF);
92 expected_inputs.push_back(dt);
94 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}),
137 DataTypeVector expected_inputs; variable
139 expected_inputs.push_back(DT_RESOURCE);
141 expected_inputs.push_back(DT_STRING_REF);
144 expected_inputs.push_back(dt);
146 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}),
[all...] |
| conditional_accumulator_base_op.h | 91 DataTypeVector expected_inputs = {}; local
92 TF_RETURN_IF_ERROR(ctx->MatchSignature(expected_inputs, {DT_STRING_REF}));
|
| lookup_table_op.cc | 662 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), variable
665 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs));
698 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), variable
700 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {}));
775 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), variable
777 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {}));
[all...] |
| barrier_ops.cc | 592 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT32}; variable
599 ctx, ctx->MatchSignature(expected_inputs, expected_outputs), callback);
|
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model_test.cc | 122 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; local
125 BuildSignatureDef(expected_inputs, expected_outputs);
133 EXPECT_EQ(expected_inputs, inputs);
152 std::unordered_set<string> expected_inputs = {"input0:0", "input1:0"}; local
154 EXPECT_EQ(expected_inputs, inputs);
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.cc | 48 Status MatchSignatureHelper(const DataTypeSlice expected_inputs,
54 if (inputs.size() != expected_inputs.size()) signature_mismatch = true;
56 if (!TypesCompatible(expected_inputs[i], inputs[i])) {
72 " expected: ", DataTypeSliceString(expected_inputs), "->",
212 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) {
213 return MatchSignatureHelper(expected_inputs, expected_outputs, input_types_,
809 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs,
816 return MatchSignatureHelper(expected_inputs, expected_outputs, inputs,
[all...] |
| op_kernel.h | 296 // If expected_inputs == inputs() and expected_outputs == output_types(),
299 Status MatchSignature(const DataTypeSlice expected_inputs,
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 782 std::vector<string> expected_inputs, test_inputs; local
787 &expected_inputs, &expected_fetches);
817 std::vector<std::pair<string, Tensor>> expected_feeds(expected_inputs.size());
819 CHECK_EQ(input_tensors.size(), expected_inputs.size());
823 expected_feeds[i] = {expected_inputs[i], input_tensors[i]};
[all...] |
