Home
last modified time | relevance | path

Searched refs:ShapeRepresentationFn (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h63 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
72 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
83 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
123 XlaCompiler::ShapeRepresentationFn shape_representation_fn;
226 const XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_device_context.h60 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
82 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
111 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_device_context.cc77 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in XlaDeviceContext()
Dxla_device.cc142 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in Metadata()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h281 ShapeRepresentationFn; typedef
308 ShapeRepresentationFn shape_representation_fn;
Dxla_compiler.cc163 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn, in BuildComputation()
1165 : ShapeRepresentationFn{}, in CompileGraph()