/external/tensorflow/tensorflow/core/framework/ |
function_testlib.h | 58 gtl::ArraySlice<FunctionDef> funcs = {}); 64 FunctionDef XTimesTwo(); 67 FunctionDef TwoDeviceTimesFive(); 70 FunctionDef TwoDeviceMult(); 73 FunctionDef TwoDeviceInputOutput(); 76 FunctionDef FuncWithListInput(); 79 FunctionDef FuncWithListOutput(); 82 FunctionDef XAddX(); 85 FunctionDef XTimesTwoInt32(); 88 FunctionDef XTimesFour() [all...] |
graph_to_functiondef.h | 25 // Converts 'graph' to a FunctionDef 'fdef', with name 'name'. 28 FunctionDef* fdef);
|
function_testlib.cc | 32 gtl::ArraySlice<FunctionDef> funcs) { 60 FunctionDef NonZero() { 76 FunctionDef IsZero() { 94 FunctionDef RandomUniform() { 119 FunctionDef XTimesTwo() { 138 FunctionDef TwoDeviceMult() { 178 FunctionDef TwoDeviceInputOutput() { 210 FunctionDef FuncWithListInput() { 228 FunctionDef FuncWithListOutput() { 246 FunctionDef XAddX() [all...] |
graph_to_functiondef_test.cc | 32 FunctionDef RemoveDebugInfo(const FunctionDef& def) { 33 FunctionDef copy = def; 40 bool EqualFunctionDef(const FunctionDef& a, const FunctionDef& b, 72 FunctionDef fdef; 75 FunctionDef fdef_expected = FunctionDefHelper::Create( 109 FunctionDef fdef; 112 FunctionDef fdef_expected = FunctionDefHelper::Create(
|
function.h | 52 // FunctionDef proto. 54 // FunctionDef my_func = FunctionDefHelper::Create( 68 // FunctionDef my_func = FunctionDefHelper::Define( 110 // Node is used to construct FunctionDef.Node using initialization 126 // Creates a FunctionDef from the given parameters. Node inputs must use 132 static FunctionDef Create( 139 // Creates a FunctionDef from the given parameters. Node inputs must use 143 static FunctionDef Create(const string& function_name, 151 static FunctionDef Define(const string& function_name, 158 static FunctionDef Define(gtl::ArraySlice<string> arg_def [all...] |
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
vectorization_utils.h | 27 // Given a MapDefun node (`map_defun_node`) in a FunctionDef (`outer_scope`) 89 Status VectorizeMapDefun(const FunctionDef& outer_scope, 91 FunctionDef** result);
|
function_utils.h | 36 // Describes a FunctionDef input tensor. In FunctionDefs, input tensor strings 58 void ReplaceReferences(const string& from, const string& to, FunctionDef* func); 64 FunctionDef* function, DataType dt); 66 // Adds a node to a FunctionDef. 70 FunctionDef* fd); 74 const FunctionDef& function); 77 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function); 81 const FunctionDef& function); 85 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function); 89 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function) [all...] |
fusion_utils.h | 59 const FunctionDef& first_function, const FunctionDef& second_function, 60 FunctionDef* fused_function, FunctionDefLibrary* library)>; 62 void MergeNodes(const FunctionDef& first_function, 63 const FunctionDef& second_function, FunctionDef* fused_function, 114 void LazyConjunctionNodes(const FunctionDef& first_function, 115 const FunctionDef& second_function, 116 FunctionDef* fused_function, 125 FunctionDef* FuseFunctions [all...] |
vectorization_utils_test.cc | 50 Status WrapFunctionWithMapDefun(const FunctionDef& inner, FunctionDef* result) { 96 Status WrapAndVectorize(const FunctionDef& fn, FunctionDefLibrary* lib, 97 FunctionDef** result) { 98 FunctionDef outer; 118 string GetRetval(const FunctionDef& function_def, int index) { 159 FunctionDef inner = FunctionDefHelper::Create( 167 FunctionDef* vectorized; 234 FunctionDef inner = FunctionDefHelper::Create( 247 FunctionDef* vectorized [all...] |
function_utils.cc | 74 // FunctionDef -> GraphDef conversion isn't really necessary in this case. 76 FunctionDef* func) { 91 FunctionDef* function, DataType dt) { 108 FunctionDef* fd) { 126 const FunctionDef& function) { 130 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { 135 const FunctionDef& function) { 139 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function) { 145 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function) { 151 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function) [all...] |
filter_fusion.cc | 38 const FunctionDef& fused_function, 81 const NodeDef* second_filter_node) -> FunctionDef* { 83 const FunctionDef* first_func = 86 const FunctionDef* second_func = function_library.Find(fun.func().name());
|
function_utils_test.cc | 45 FunctionDef outer = FunctionDefHelper::Create( 57 FunctionDef function = test::function::XTimesTwo(); 64 FunctionDef function = test::function::XTimesTwo(); 71 FunctionDef function = test::function::XTimesTwo(); 78 FunctionDef function = test::function::XTimesTwo(); 84 FunctionDef function = test::function::XTimesTwo(); 92 FunctionDef function = test::function::XTimesTwo(); 100 FunctionDef function = test::function::XTimesTwo(); 106 FunctionDef function = test::function::XTimesTwo(); 112 FunctionDef function = test::function::XTimesTwo() [all...] |
fusion_utils.cc | 50 string GetOutputNode(const FunctionDef& function, int output_idx) { 56 string& GetMutableOutputNode(FunctionDef* function, int output_idx) { 115 void RenameFunctionNodes(const FunctionDef& first_function, 143 StringCollection GetFunctionInputs(const FunctionDef& function) { 250 StringCollection GetFunctionOutputs(const FunctionDef& function) { 260 FunctionDef* CreateFalsePredicate( 297 void MergeNodes(const FunctionDef& first_function, 298 const FunctionDef& second_function, FunctionDef* fused_function, 376 void LazyConjunctionNodes(const FunctionDef& first_function [all...] |
/external/tensorflow/tensorflow/core/ops/ |
math_grad.cc | 25 Status GradForUnaryCwise(FunctionDef* g, std::vector<FDH::Node> nodes) { 43 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) { 53 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) { 62 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) { 75 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) { 87 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) { 101 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) { 116 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) { 126 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) { 136 Status LogGrad(const AttrSlice& attrs, FunctionDef* g) [all...] |
nn_grad.cc | 25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { 50 Status LogSoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { 75 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) { 93 Status Relu6Grad(const AttrSlice& attrs, FunctionDef* g) { 111 Status CrossEntropyGrad(const AttrSlice& attrs, FunctionDef* g) { 140 Status Conv2DGrad(const AttrSlice& attrs, FunctionDef* g) { 176 Status MaxPoolGrad(const AttrSlice& attrs, FunctionDef* g) { 207 Status AvgPoolGrad(const AttrSlice& attrs, FunctionDef* g) { 233 Status MaxPoolGradGrad(const AttrSlice& attrs, FunctionDef* g) { 264 Status BiasAddGrad(const AttrSlice& attrs, FunctionDef* g) [all...] |
array_grad.cc | 33 Status ReshapeGrad(const AttrSlice& attrs, FunctionDef* g) { 54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) { 73 Status IdentityGrad(const AttrSlice& attrs, FunctionDef* g) { 92 Status PackGrad(const AttrSlice& attrs, FunctionDef* g) { 118 Status UnpackGrad(const AttrSlice& attrs, FunctionDef* g) { 142 Status ConcatGradHelper(const AttrSlice& attrs, FunctionDef* g, 216 Status ConcatGrad(const AttrSlice& attrs, FunctionDef* g) { 220 Status ConcatGradV2(const AttrSlice& attrs, FunctionDef* g) { 227 Status SplitGrad(const AttrSlice& attrs, FunctionDef* g) { 247 Status SplitVGrad(const AttrSlice& attrs, FunctionDef* g) [all...] |
functional_grad.cc | 24 Status MapAccumulateGrad(const AttrSlice& attrs, FunctionDef* ret) {
|
/external/tensorflow/tensorflow/core/common_runtime/ |
function_testlib.h | 26 FunctionDef FindDevice(); 42 FunctionDef BlockingOpFn();
|
function.h | 76 FunctionDef fdef; 85 FunctionBody(const FunctionDef& f, DataTypeSlice arg_types, 167 // a) control returns (`control_ret` field in FunctionDef) 168 // b) data returns (`ret` field in FunctionDef) 221 // side-effects that must always execute (see `control_ret` in FunctionDef). 284 // Instantiates FunctionDef into a graph. Set *fbody to point to the 285 // FunctionBody that holds the instantiated FunctionDef. 287 const FunctionDef& fdef, const AttrSlice& attrs,
|
/external/tensorflow/tensorflow/core/util/ |
dump_graph.h | 45 // Similar to DumpGraphDefToFile, but dumps a function as a FunctionDef text 47 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef,
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
functions.h | 83 // FunctionDef uses different connectivity encoding for the function body nodes, 84 // then a GraphDef (see function.proto for details). Input name in FunctionDef 86 // GraphDef), we need to expand it when converting from FunctionDef to GraphDef, 94 // Expands input encoded in FunctionDef format (name[:output][:position]) into 99 // Updates Node inputs from FunctionDef to GraphDef format. 105 // FunctionDef built from GrapplerFunctionItem is always specialized to its 110 // FunctionDef input format (name[:output][:position]) using registered input 115 // Updates Node inputs from GraphDef to FunctionDef format. 182 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&, 212 bool HasParametrizedType(const FunctionDef& func) [all...] |
/external/tensorflow/tensorflow/compiler/jit/ |
create_xla_launch_op_test.cc | 38 // Create a FunctionDef that takes one resource and one regular param 39 FunctionDef XTimesY() { 58 void Init(const std::vector<FunctionDef>& flib) { 95 FunctionDef fdef = XTimesY(); 120 FunctionDef fdef = XTimesY(); 133 FunctionDef fdef = XTimesY();
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
test_util.cc | 26 const FunctionDef* fdef = library.Find(name);
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
function_api_info.cc | 29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { 102 bool IsSameSignature(const FunctionDef& f1, const FunctionDef& f2, 124 const std::vector<const FunctionDef*>& equiv_funcs, 146 const std::unordered_map<string, std::vector<const FunctionDef*>>& 158 std::unordered_map<string, std::vector<const FunctionDef*>> infer_funcs; 159 std::unordered_map<string, std::vector<const FunctionDef*>> fwd_funcs; 160 std::unordered_map<string, std::vector<const FunctionDef*>> bwd_funcs;
|
/external/tensorflow/tensorflow/core/kernels/data/ |
map_dataset_op_test.cc | 93 std::vector<FunctionDef> input_func_lib) 106 std::vector<FunctionDef> func_lib; 161 std::vector<FunctionDef>{test::function::XTimesTwo()}), 163 std::vector<FunctionDef>{test::function::XAddX()}), 166 std::vector<FunctionDef>{test::function::XTimesTwo(), 172 FunctionDef func_def = test::function::XTimesTwo(); 199 FunctionDef func_def = test::function::XTimesTwo(); 227 FunctionDef func_def = test::function::XTimesTwo(); 278 FunctionDef func_def = test::function::XTimesTwo(); 311 FunctionDef func_def = test::function::XTimesTwo() [all...] |