Searched refs:tensor_array_size (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 63 TF_RET_CHECK(resource->tensor_array_size() >= 0) in MaybeInitializeTensorArray()
64 << resource->name() << " size " << resource->tensor_array_size(); in MaybeInitializeTensorArray()
83 ta_shape.AddDim(resource->tensor_array_size()); in MaybeInitializeTensorArray()
120 shape->InsertDim(0, resource->tensor_array_size()); in GetTensorArrayShape()
514 ta_shape.AddDim(resource->tensor_array_size()); in Compile()
518 ctx, lengths.size() == resource->tensor_array_size(), in Compile()
521 lengths.size(), " vs. ", resource->tensor_array_size(), ")")); in Compile()
555 static_cast<int32>(var->tensor_array_size()); in Compile()
|
| D | while_op.cc | 65 arg.tensor_array_size = resource->tensor_array_size(); in MakeXlaCompilerArgumentsFromInputs()
|
| D | stack_ops.cc | 73 stack_shape.AddDim(resource->tensor_array_size()); in MaybeInitializeStack()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_resource.h | 42 int64 tensor_array_size,
126 int64 tensor_array_size() const { return tensor_array_size_; } in tensor_array_size() function
|
| D | xla_compiler.cc | 69 if (std::tie(kind, resource_kind, type, name, initialized, tensor_array_size, in operator ==()
72 other.initialized, other.tensor_array_size, in operator ==()
251 if (arg.tensor_array_size < 0) { in XLAShapeForArgument()
256 shape.AddDim(arg.tensor_array_size); in XLAShapeForArgument()
268 if (arg.tensor_array_size < 0) { in XLAShapeForArgument()
273 shape.AddDim(arg.tensor_array_size); in XLAShapeForArgument()
457 /*tensor_array_size=*/arg.tensor_array_size, in BuildArguments()
|
| D | xla_resource.cc | 31 int64 tensor_array_size, in XlaResource() argument
40 tensor_array_size_(tensor_array_size) { in XlaResource()
|
| D | xla_context.cc | 112 int64 tensor_array_size, const std::set<string>& tensor_array_gradients, in CreateResource() argument
116 handle, tensor_array_size, tensor_array_gradients)); in CreateResource()
|
| D | xla_context.h | 83 int64 tensor_array_size,
|
| D | xla_compiler.h | 137 int64 tensor_array_size = -1; member
|
| D | xla_compiler_test.cc | 477 args[0].tensor_array_size = 2; in TEST_F()
543 args[0].tensor_array_size = 2; in TEST_F()
575 args[0].tensor_array_size = 2; in TEST_F()
|