| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_jit_compiled_cpu_function.cc | 104 std::vector<const xla::Shape*> arg_shapes; local
105 arg_shapes.reserve(program_shape->parameters_size());
107 arg_shapes.push_back(&program_shape->parameters(i));
114 client->Compile(computation, arg_shapes, build_options));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_cpu.cc | 108 std::vector<xla::Shape> arg_shapes; variable
115 arg_shapes.push_back(std::move(arg_shape));
122 xla_shape, arg_shapes);
125 xla_shape, arg_shapes);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.h | 100 std::vector<std::pair<DataType, std::vector<int64>>> arg_shapes; member in struct:tensorflow::XlaCompilationCache::Signature
|
| encapsulate_util.cc | 315 std::map<int, InferredShape> arg_shapes; local
318 InferShapes(g, arg_shapes, /*fnlib_def=*/nullptr, &shape_info));
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_defun_op.cc | 138 const std::vector<TensorShape> arg_shapes; member in struct:tensorflow::data::__anon45310::MapDefunOp::ComputeOptions
152 std::vector<TensorShape> arg_shapes, int64 batch_size,
156 arg_shapes(std::move(arg_shapes)),
196 compute_opts_->arg_shapes.at(index));
287 std::vector<TensorShape> arg_shapes; local
288 arg_shapes.reserve(arguments.size());
291 arg_shapes.push_back(arguments[i].shape());
292 arg_shapes.at(i).RemoveDim(0);
296 ctx, arguments, captured_inputs, std::move(arg_shapes), batch_size
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| numpy_bridge.cc | 157 Safe_PyObjectPtr arg_shapes = make_safe(PyTuple_New(shape.parameters_size())); local
159 PyTuple_SET_ITEM(arg_shapes.get(), i,
165 PyTuple_Pack(2, arg_shapes.release(), result_shape.release()));
|
