Home
last modified time | relevance | path

Searched defs:input_types (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/accuracy/
Dutils.h32 std::vector<DataType> input_types; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc79 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLibdeviceMathCall()
122 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
145 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitMathCall()
286 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceFunctionCall()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc41 DataTypeVector input_types({DT_FLOAT, DT_FLOAT}); in TEST_F() local
55 DataTypeVector input_types({DT_INT32, DT_INT32}); in TEST_F() local
112 DataTypeVector input_types{DT_FLOAT}; in BuildRemoteFusedGraphExecuteOp() local
Dscoped_allocator_ops_test.cc90 const DataTypeVector& input_types) { in PrepOp()
Dset_kernels.cc336 SetOperationOp(OpKernelConstruction* ctx, InputTypes input_types) in SetOperationOp()
/external/tensorflow/tensorflow/core/ops/
Dremote_fused_graph_ops_test.cc31 DataTypeVector input_types; in TEST() local
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc303 std::vector<string> input_types = in ReadTocoFlagsFromCommandLineFlags() local
Dargs.h178 Arg<string> input_types; member
/external/tensorflow/tensorflow/python/framework/
Dfunction.py93 def __init__(self, *input_types, **kwargs):
208 input_types, argument
583 def instantiate(self, input_types): argument
Dfunc_graph.py360 input_types=None, argument
388 input_types=None, argument
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc112 DataTypeVector input_types; in InsertLogging() local
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h164 DataTypeVector input_types, in AddDatasetInputFromArray()
Ddataset_test_base.cc231 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput()
/external/vixl/tools/test_generator/
Dparser.py54 def __init__(self, operand_types, operand_variants, input_types, argument
Dgenerator.py252 def GenerateInputs(self, input_types): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization_test.cc431 std::vector<DataType> input_types({DT_INT64}); in TEST() local
451 std::vector<DataType> input_types({DT_INT64}); in TEST() local
471 std::vector<DataType> input_types({DT_INT64}); in TEST() local
Dmap_vectorization.cc356 DataTypeVector input_types; in AddBranch() local
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 input_types, algebraic_properties, const_expr): argument
107 def opcode(name, output_size, output_type, input_sizes, input_types, argument
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show_multi.cc210 std::vector<string> input_types; in FormatInputShapes() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc116 DataTypeVector input_types; in BuildFusedGraphDef() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_while_test.cc82 std::vector<DataType> input_types = {DT_INT32}; in WhileWithIfBody() local
Ddirect_session.h152 DataTypeVector input_types; member
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc48 const DataTypeVector input_types; member
175 const DataTypeVector& Node::input_types() const { return props_->input_types; } in input_types() function in tensorflow::Node
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc426 DataTypeSlice input_types = {}) { in ExpectSuccess()
498 DataTypeSlice input_types = {}) { in GetKernelClassName()
Dop_kernel.h159 const DataTypeVector& input_types() const { return input_types_; } in input_types() function
315 const DataTypeSlice& input_types() const { return input_types_; } in input_types() function

12