/external/tensorflow/tensorflow/contrib/quantize/python/ |
D | quantize_parameterized_test.py | 82 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 …]
|
D | quantize_graph_test.py | 523 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)
|
D | fold_batch_norms_test.py | 779 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/ |
D | queue_op.cc | 90 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()
|
D | sparse_conditional_accumulator_op.cc | 76 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()
|
D | conditional_accumulator_op.cc | 76 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()
|
D | lookup_table_init_op.cc | 55 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()
|
D | conditional_accumulator_base_op.h | 94 DataTypeVector expected_inputs = {}; in SetAccumulatorHandle() local 95 TF_RETURN_IF_ERROR(ctx->MatchSignature(expected_inputs, {DT_STRING_REF})); in SetAccumulatorHandle()
|
D | lookup_table_op.cc | 794 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()
|
D | barrier_ops.cc | 591 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/ |
D | freeze_saved_model_test.cc | 277 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/ |
D | client_stub_spec.rb | 984 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/ |
D | op_kernel.cc | 57 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()
|
D | op_kernel.h | 331 Status MatchSignature(const DataTypeSlice expected_inputs, 1131 Status MatchSignature(const DataTypeSlice expected_inputs,
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 870 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()
|